//
//  Problem28.swift
//  LeetCode 28 实现 strStr()
//
//  Created by 武侠 on 2020/7/25.
//  Copyright © 2020 zhulongiMac02. All rights reserved.
//

import UIKit

/*
 实现 strStr() 函数。

 给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回  -1。
 示例 1:

 输入: haystack = "hello", needle = "ll"
 输出: 2
 示例 2:

 输入: haystack = "aaaaa", needle = "bba"
 输出: -1
 说明:
 当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。
 对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。

 */
class Problem28: NSObject {
    func solution() {
        var lists = [["hello", "ll"], ["aaaaa", "bba"], ["aaaaa", "a"], ["", "a"], ["", ""], ["a", "a"], ["abb", "abaaa"]]
        for list in lists {
            print(strStr(list[0], list[1]))
        }
    }
    
    // 遍历第一个字符串, 然后遍历第二哥字符串,看看是否有相等的
    func strStr(_ haystack: String, _ needle: String) -> Int {
        if haystack.count == 0, needle.count == 0 {
            return 0
        }
        
        if haystack.count == 0 {
            return -1
        }
        if needle.count == 0 {
            return 0
        }
        
        if needle.count > haystack.count {
            return -1
        }
        
        var i = haystack.startIndex
        let endIndex = haystack.index(haystack.endIndex, offsetBy: 1-(needle.count))
        while i != endIndex {
            var ii = i
            var j = needle.startIndex
            var n: Int = 0
            while j != needle.endIndex {
                if haystack[ii] == needle[j] {
                    ii = haystack.index(ii, offsetBy: 1)
                    j = needle.index(j, offsetBy: 1)
                    n += 1
                } else {
                    break
                }
            }
            if j == needle.endIndex {
                return haystack.distance(from: haystack.startIndex, to: i)
            }
            i = haystack.index(i, offsetBy: 1)
        }
        return -1
    }
}
